#include <iostream>
#include "captcha_recognize_new.h"


int main() {
#if 1
    //const std::string train_path = "D:/CPP/captcha_lstm_cpp/dataset/verify_code_xuexin_all/train";
    //const std::string valid_path = "D:/CPP/captcha_lstm_cpp/dataset/verify_code_xuexin_all/valid";

    const std::string train_path = "C:/Users/SongpingWang/Desktop/verify_code_xuexin/verify_code_xuexin_generate_train2";
    const std::string valid_path = "C:/Users/SongpingWang/Desktop/verify_code_xuexin/verify_code_xuexin_generate_valid2";
#else
    //const std::string train_path = "E:/Desktop/verify_code_xuexin_all/train";
    //const std::string valid_path = "E:/Desktop/verify_code_xuexin_all/valid";

    const std::string train_path = "E:/Desktop/verify_code_xuexin/verify_code_xuexin_generate_train2";
    const std::string valid_path = "E:/Desktop/verify_code_xuexin/verify_code_xuexin_generate_valid2";
#endif
    const std::string save_path = "./";
    const int BATCH_SIZE = 128;
    const int TOTAL_EPOCH = 150;
    captcha_recognize captcha_recognize;
    captcha_recognize.Start(BATCH_SIZE, TOTAL_EPOCH, train_path, valid_path, save_path);

    return 0;
}
